﻿// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// Copyright (c) 2011, Dr. Masroor Ehsan
// All rights reserved.
// 
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//     * Redistributions of source code must retain the above copyright
//       notice, this list of conditions and the following disclaimer.
//     * Redistributions in binary form must reproduce the above copyright
//       notice, this list of conditions and the following disclaimer in the
//       documentation and/or other materials provided with the distribution.
//     * Neither the name of the author nor the
//       names of its contributors may be used to endorse or promote products
//       derived from this software without specific prior written permission.
// 
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
// WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
// DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
// DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
// (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
// ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
// 
// $Author:$
// $Id:$
// $Rev:$
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *

using CuttingEdge.Conditions;
using MaxTrader.Core;

namespace MaxTrader.BusinessObjects
{
    /// <summary>
    ///   Representation of a cash account in a portfolio
    /// </summary>
    public class CashAccount : AbstractAccount
    {
        protected CashAccount()
        {
            AccountType = AccountType.CashAccount;
        }

        public CashAccount(string accountName, Currency currency) : this()
        {
            AccountName = accountName;
            Currency = currency;
            Capital = new Money(0m);
        }

        /// <summary>
        ///   Create a new cash account with the Bangladeshi currency.
        /// </summary>
        public CashAccount(string accountName) : this()
        {
            AccountName = accountName;
            Currency = Currency.Bdt;
            Capital = new Money(0m);
        }

        public virtual Money Capital { get; protected set; }

        public virtual void AddCapital(decimal amount)
        {
            Capital += amount;
        }

        public override Money GetValue()
        {
            return Capital;
        }

        public override Money GetValue(IQuoteSource quoteSource, TradingDate date)
        {
            return Capital;
        }

        public override void ApplyTransaction(AbstractTransaction transaction)
        {
            Condition
                .Requires(transaction, "transaction")
                .IsOfType(typeof (CashTransaction));
            Condition
                .Requires(transaction.Amount.Currency, "transaction.Amount.Currency")
                .Evaluate(c => c.Equals(Currency));

            var cashTrans = transaction as CashTransaction;

            switch (cashTrans.Category)
            {
                case TransactionCategory.Deposit:
                case TransactionCategory.Interest:
                case TransactionCategory.Dividend:
                    Capital = Capital + cashTrans.Amount;
                    break;

                case TransactionCategory.Withdrawal:
                case TransactionCategory.Fee:
                    Capital = Capital - cashTrans.Amount;
                    break;

                case TransactionCategory.Accumulate:
                    Condition.Requires(cashTrans.TradeCost.Currency).Evaluate(c => c.Equals(Currency));
                    Capital -= cashTrans.Amount;
                    Capital -= cashTrans.TradeCost;
                    break;

                case TransactionCategory.Reduce:
                    Capital += cashTrans.Amount;
                    Capital -= cashTrans.TradeCost;
                    break;

                case TransactionCategory.Transfer:
                    // Are we transfering to or from this account?
                    if (cashTrans.CashAccount == this)
                    {
                        Capital = Capital - cashTrans.Amount; // from
                    }
                    else
                    {
                        Capital = Capital + cashTrans.Amount; // to
                    }
                    break;
            }
        }

        public override object Clone()
        {
            return new CashAccount(AccountName, Currency);
        }

        public override bool Equals(AbstractAccount other)
        {
            if (other is CashAccount)
            {
                var cashOther = (CashAccount) other;
                return (cashOther.AccountName == AccountName &&
                        cashOther.Currency == Currency &&
                        cashOther.GetValue() == Capital);
            }

            return false;
        }
    }
}